from ultralytics import YOLO

if __name__ == '__main__':
    model = YOLO('yolov8n-cls-resnet50.yaml')
    # results = model.train(data='datasets/standford_dogs_yolo', epochs=10, imgsz=224)
    results = model.train(data='datasets/cub200_2011_yolo', epochs=10, imgsz=224)